#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@file    device.py
@brief
@details
@author  Shivelino
@date    2023-12-23 19:08
@version 0.0.1

@par Copyright(c):
@par todo:
@par history:
"""
import torch


def get_device():
    """get auto device. GPU first."""
    return torch.device("cuda" if torch.cuda.is_available() else "cpu")
